from pymongo.collection import Collection


def get_fast_count(collection: Collection, query=None):
    """
    获取集合中文档的快速计数

    Args:
        collection: MongoDB集合对象
        query: 查询条件，为None时使用估算计数

    Returns:
        int: 文档数量
    """
    if not query:
        try:
            return collection.estimated_document_count()
        except Exception:
            return collection.count_documents({})
    else:
        return collection.count_documents(query)
